# 统计IEMOCAP一共多少个音频文件
import os

def count_wav_files(directory):
    # 变量用于存储找到的 .wav 文件数量
    wav_count = 0

    # os.walk 会递归遍历目录及其子目录
    for root, dirs, files in os.walk(directory):
        # 遍历当前目录中的文件
        for file in files:
            # 判断文件扩展名是否是 .wav
            if file.lower().endswith('.wav'):
                wav_count += 1

    return wav_count

# 指定要搜索的目录路径
directory = '/hdd/data/swp/IEMOCAP/IEMOCAP_full_release/Session5/sentences/wav'

# 获取并打印总共的 .wav 文件数量
total_wav_files = count_wav_files(directory)
print(f"Total .wav files found: {total_wav_files}")
